tensor([[0.7200, 0.7200, 0.7200, 0.7200, 0.7200],
        [0.2800, 0.2800, 0.2800, 0.2800, 0.2800]], grad_fn=<RoundBackward1>)